package com.epam.lissina.factory;

import java.util.ArrayList;

import com.epam.lissina.entity.MedicalTour;
import com.epam.lissina.entity.TravelAgency;
import com.epam.lissina.entity.TravelTour;
import com.epam.lissina.entity.VacationTour;
import com.epam.lissina.enums.TypeOfTransport;

public class AgencyFactory {

	public static ArrayList<TravelAgency> build() {

		ArrayList<TravelAgency> res = new ArrayList<TravelAgency>();

		TravelAgency travelAgency1 = new TravelAgency(1, "Voyage");
		travelAgency1.addTour(new TravelTour(TypeOfTransport.AUTO, 128923, 6, 3,
				TypeOfTransport.MOTO));
		travelAgency1.addTour(new MedicalTour(TypeOfTransport.PLANE, 143200, 10,
				14500, 2));
		travelAgency1.addTour(new VacationTour(TypeOfTransport.TRAIN, 142700, 10,
				5));
		res.add(travelAgency1);

		TravelAgency travelAgency2 = new TravelAgency(2, "Sayahat");
		travelAgency2.addTour(new TravelTour(TypeOfTransport.PLANE, 340023, 5, 2,
				TypeOfTransport.MOTO));
		travelAgency2.addTour(new MedicalTour(TypeOfTransport.SHIP, 280200, 12,
				0, 2));
		travelAgency2.addTour(new VacationTour(TypeOfTransport.TRAIN, 132700, 8,
				5));
		res.add(travelAgency2);

		return res;
	}
}